# Copyright (c) 2018 Presto Labs Pte. Ltd.
# Author: jaewon

from coin.exchange.huobi_futures.ws.subscriber import HuobiFuturesSubscriber
from coin.strategy.mm.feed_builder import HuobiFuturesFlowBookBuilder
from coin.strategy.mm.base.feed import BaseFeedSubsystem
from coin.strategy.mm.subscription import SubscriptionRequest


class HuobiFuturesFeedSubsystem(BaseFeedSubsystem):
  FeedSubReq = SubscriptionRequest('Futures', 'Huobi', None)
  FeedSubscriber = HuobiFuturesSubscriber
  FlowBookBuilder = HuobiFuturesFlowBookBuilder

  def _get_subscription(self):
    return [product.native_subscription_symbol for product in self._products]
